package com.linyaonan.leetcode.sword_to_offer._35;

import java.util.HashMap;

/**
 * 请实现 copyRandomList 函数，复制一个复杂链表。在复杂链表中，每个节点除了有一个 next 指针指向下一个节点，还有一个 random 指针指向链表中的任意节点或者 null。
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/fu-za-lian-biao-de-fu-zhi-lcof
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
class Solution {
    /**
     * 尝试直接进行copy实现
     * @param head
     * @return
     */
    public Node copyRandomList(Node head) {
        if (head == null) {
            return head;
        }

        Node o1 = head;
        HashMap<Node, Node> map = new HashMap<>();
        while (o1 != null) {
            map.put(o1, new Node(o1.val));
            o1 = o1.next;
        }

        Node o2 = head;
        Node r = o2;
        while (o2 != null) {
            Node node = map.get(o2);
            node.next = map.get(o2.next);
            node.random = map.get(o2.random);
            o2 = o2.next;
        }

        return map.get(r);
    }
}